package com.example.lambda.stream;

import com.example.lambda.test4.Employee;
import com.example.lambda.test4.EmployeeData;
import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Stream;

/**
 * @author readpage
 * @create 2022-10-22 16:30
 *
 * 测试Stream的中间操作
 */
public class StreamApiTest2 {
    //1-筛选与切片

    /**
     * filter(Predicate p) 接受Lambda,从流中排除某些元素
     * distinct() 筛选，通过流所生成元素的hashCode()和equals()去除重复元素
     * limit(long maxSize) 截断流，使其元素不超过给定数量
     * skip(long n) 跳过元素，返回一个扔掉了前n个元素的流。若流中元素不足n个，则返回一个空流。与limit(n)互补。
     */
    @Test
    public void test() {
        List<Employee> list = EmployeeData.getEmployees();
        Stream<Employee> stream = list.stream();
        //查询员工表中薪资大于7000的员工信息
        stream.filter(e -> e.getSalary() > 7000).forEach(System.out::println);
        System.out.println("**********");
        // limit(n)-
        list.stream().limit(3).forEach(System.out::println);
        System.out.println("**********");
        list.stream().skip(3).forEach(System.out::println);
        System.out.println("***********");
        list.add(new Employee(1010, "刘强东", 40, 8000));
        list.add(new Employee(1010, "刘强东", 40, 8000));
        list.add(new Employee(1010, "刘强东", 40, 8000));
        list.add(new Employee(1010, "刘强东", 40, 8000));
//        System.out.println(list);
        list.stream().distinct().forEach(System.out::println);
    }

    //映射

    /**
     * map(Function f)  接收一个函数作为参数，该函数会被应用到每个元素上，并将其映射成一个新的元素
     * mapToDouble(ToDoubleFunction f)  接收一个函数作为参数，该函数会被应用到每个元素上，产生一个新的DoubleStream
     * mapToInt(ToIntFunction f)    接收一个函数作为参数，该函数会被应用到每个元素上，产生一个新的IntStream
     * mapToLong(ToLongFunction f)  接收一个函数作为参数，该函数会被应用到每个元素上，产生一个新的LongStream。
     * flatMap(Function f)  接收一个函数作为参数，将流中的每一个值都换成另一个流，然后把所有流连接成一个流
     */
    @Test
    public void test2() {
        List<String> list = Arrays.asList("aa", "bb", "cc");
        list.stream().map(String::toUpperCase).forEach(System.out::println);

        System.out.println("**********");
        List<Employee> employees = EmployeeData.getEmployees();
        Stream<String> stringStream = employees.stream().map(Employee::getName);
        stringStream.filter(name -> name.length() > 3).forEach(System.out::println);
    }

    @Test
    public void test3() {
        List<Integer> list = new ArrayList<>(Arrays.asList(1, 2, 3));
        List<Integer> list2 = new ArrayList<>(Arrays.asList(4, 5, 6));

        list.addAll(list2);

        System.out.println(list);
    }

    //3-排序
    @Test
    public void test4() {
        // sorted()--自然排序
        List<Integer> list = Arrays.asList(12, 43, 65, 34, 87, 0, -98, 7);
        list.stream().sorted().forEach(System.out::println);

        // sorted(Comparator com)--定制排序
        System.out.println("********************");
        List<Employee> employees = EmployeeData.getEmployees();
        // 根据id排序
        employees.stream().sorted(Comparator.comparing(Employee::getId)).forEach(System.out::println);
        System.out.println("******************");
        // 逆序
        employees.stream().sorted(Comparator.comparing(Employee::getId).reversed()).forEach(System.out::println);
        System.out.println("******************");
        employees.stream().sorted((e1, e2) -> {
            int ageValue = Integer.compare(e1.getAge(), e2.getAge());
            if (ageValue == 0) {
                return Double.compare(e1.getSalary(), e2.getSalary());
            } else{
                return ageValue;
            }
        }).forEach(System.out::println);
    }
}
